package com.xyhuo.leetcode;

import java.util.*;

public class ContainsNearbyDuplicate {

    public boolean containsNearbyDuplicate(int[] nums, int k) {
        Map<Integer, List<Integer>> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            if (map.containsKey(nums[i])) {
                map.get(nums[i]).add(i);
            } else {
                List<Integer> list = new ArrayList<>();
                list.add(i);
                map.put(nums[i], list);
            }
        }

        for (Map.Entry<Integer, List<Integer>> entry : map.entrySet()) {
            if (entry.getValue().size() > 1) {
                List<Integer> list = entry.getValue();

                for (int i = 1; i < list.size(); i++) {
                    if (list.get(i) - list.get(i - 1) <= k) {
                        return true;
                    }
                }
            }
        }
        return false;
    }
}
